EasyPPO: Stabilizing the Critic Is Key
EasyPPO:稳定 Critic 是 PPO 稳定训练的关键
🧭 RLVR / PPO 训练稳定性优化 ★ 深度解读
University of California, Berkeley · Princeton University
通过 actor-only 截断过滤、按回报方差加权 critic 损失和较小 critic mini-batch,稳定 PPO 训练并全面优于现有方法
📌 工作参考价值(4 点)
- 若在搜索/推荐 Agent 或生成式排序中用 PPO/RLVR 训练 LLM,不要同时对 actor 和 critic 过滤超长 rollout:截断样本保留给 critic 学习,能避免策略偏向“早点截断也能拿到的条件奖励”,同时 actor 侧只过滤截断样本仍可降低噪声。
- 当不同 query / 用户组 / 问题上的 reward 方差差异很大(如连续奖励:转化金额、延迟、GPU kernel 加速比等),对 critic loss 使用 prompt 组内回报标准差的倒数加权,可以在现有分组采样 rollout 的框架下低成本实现,防止少数高方差 prompt 主导 critic 更新。
- 工程上 critic mini-batch 不必等于整个 rollout batch;默认切成 4 个 mini-batch 并做 gradient clipping,能限制 outlier 影响范围。若改变 mini-batch 数,记得按 η ∝ 1/√K 调整 critic 学习率。
- 训练监控可以看 critic 的 explained variance 和 clip 前 gradient norm;持续大幅波动或 EV 突然下降通常是 critic 失效的前兆,可作为在线崩溃预警。